package ahpu.piconet.util;

import java.io.OutputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.List;

import javax.servlet.http.HttpServletResponse;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
import org.apache.poi.hssf.usermodel.HSSFFont;
import org.apache.poi.hssf.usermodel.HSSFHeader;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.hssf.util.HSSFColor;

import ahpu.piconet.domin.Data;

/**
 * 利用开源组件POI导出EXCEL文档
 * @param <T> 应用泛型，代表任意一个符合javabean风格的类
 */
public class ExcelUtil<T> {
	//excel中sheet的名称
	private String sheetName ="报表内容";
        public ExcelUtil() {
		super();
	}
	public ExcelUtil(String sheetName) {
		super();
		this.sheetName = sheetName;
	}

    /**
     * 这是一个通用的方法，利用了JAVA的反射机制，可以将放置在JAVA集合中并且符号一定条件的数据以EXCEL 的形式输出到指定IO设备上
     * @param headers
     *            表格属性列名数组
     * @param dataSet
     *            需要显示的数据集合,集合中一定要放置符合javabean风格的类的对象。
     * @param response
     *            与输出设备关联的流对象，可以将EXCEL文档导出到本地文件或者网络中
     */
    @SuppressWarnings({ "unchecked", "deprecation" })
    public void setExcelContents(String[] headers,List<Data> dataSet,HttpServletResponse response) {
    	//创建一个新的Excel  
        HSSFWorkbook workBook = new HSSFWorkbook();  
        //创建sheet页  
        HSSFSheet sheet = workBook.createSheet(sheetName);
        // 设置表格默认列宽度为25个字节
        sheet.setDefaultColumnWidth((short) 25);
        //创建header页  
        HSSFHeader header = sheet.getHeader();  
        //设置标题居中  
        header.setCenter("标题");  
        
        // 生成一个样式
        HSSFCellStyle style = workBook.createCellStyle();
        // 设置这些样式
        style.setFillForegroundColor(HSSFColor.SKY_BLUE.index);
        style.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
        style.setBorderBottom(HSSFCellStyle.BORDER_THIN);
        style.setBorderLeft(HSSFCellStyle.BORDER_THIN);
        style.setBorderRight(HSSFCellStyle.BORDER_THIN);
        style.setBorderTop(HSSFCellStyle.BORDER_THIN);
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        // 生成一个字体
        HSSFFont font = workBook.createFont();
        font.setColor(HSSFColor.VIOLET.index);
        font.setFontHeightInPoints((short) 12);
        font.setBoldweight(HSSFFont.BOLDWEIGHT_BOLD);
        // 把字体应用到当前的样式
        style.setFont(font);
        
        // 生成并设置另一个样式
        HSSFCellStyle style2 = workBook.createCellStyle();
        style2.setFillForegroundColor(HSSFColor.LIGHT_YELLOW.index);
        style2.setFillPattern(HSSFCellStyle.SOLID_FOREGROUND);
        style2.setBorderBottom(HSSFCellStyle.BORDER_THIN);
        style2.setBorderLeft(HSSFCellStyle.BORDER_THIN);
        style2.setBorderRight(HSSFCellStyle.BORDER_THIN);
        style2.setBorderTop(HSSFCellStyle.BORDER_THIN);
        style2.setAlignment(HSSFCellStyle.ALIGN_CENTER);
        style2.setVerticalAlignment(HSSFCellStyle.VERTICAL_CENTER);
        // 生成另一个字体
        HSSFFont font2 = workBook.createFont();
        font2.setBoldweight(HSSFFont.BOLDWEIGHT_NORMAL);
        // 把字体应用到当前的样式
        style2.setFont(font2);
        
        //设置第一行为Header  
        HSSFRow row = sheet.createRow(0);  
        //设置第一行的内容和风格
        for(int i=0;i<headers.length;i++){
        	 HSSFCell cell = row.createCell(i);
             cell.setCellStyle(style);
             cell.setCellValue(headers[i]);
        }
        //设置数据
        if(dataSet != null && !dataSet.isEmpty()) {  
            for(int i = 0; i < dataSet.size(); i++) {  
            	Data datas = dataSet.get(i);  
            	//生成一行
                row = sheet.createRow(i + 1);  
                //设置内容和风格
                for(int j=0;j<headers.length;j++){
               	    HSSFCell cell = row.createCell(j);
                    cell.setCellStyle(style2);
                    switch(j){
	                    case 0:
	                    	cell.setCellValue(datas.getId());
	                    	break;
	                    case 1:
	                    	cell.setCellValue(datas.getElectricity());
	                    	break;
	                    case 2:
	                    	cell.setCellValue(datas.getVoltage());
	                    	break;
	                    case 3:
	                    	cell.setCellValue(datas.getTimer());
	                    	break;
	                    }
               }
            }  
        }  
        //通过Response把数据以Excel格式保存  
        response.reset();  
        response.setContentType("application/msexcel;charset=UTF-8");
        //时间格式
        SimpleDateFormat dateformat=new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
		String time=dateformat.format(new Date());
        try {  
            response.addHeader("Content-Disposition", "attachment;filename=\""  
                    + new String((sheetName+"_" + time+".xls").getBytes("GBK"),  
                            "ISO8859_1") + "\"");  
            OutputStream out = response.getOutputStream();  
            workBook.write(out); 
            out.flush();  
            out.close();  
        } catch (Exception e) {  
            e.printStackTrace();  
        }
    }
    public String getSheetName() {
		return sheetName;
	}
	public void setSheetName(String sheetName) {
		this.sheetName = sheetName;
	}
}